import torchvision
from torch import nn

vgg16_true = torchvision.models.vgg16(pretrained=True)
vgg16_false = torchvision.models.vgg16(pretrained=False)

rain_data = torchvision.datasets.CIFAR10("../../dataSet",
                                          transform=torchvision.transforms.ToTensor(),
                                          download=True,
                                          train=False)
# tian jia
# vgg16_true.add_module('add_linear',nn.Linear(1000,10))
vgg16_true.classifier.add_module('add_linear',nn.Linear(1000,10))
print(vgg16_true)

#xiu gai
vgg16_false.classifier[6] = nn.Linear(4096,10)
print(vgg16_false)

